/**
 *
 */
package com.totyu.bpo.raffle.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;

import com.totyu.bpo.raffle.consts.RaffleConstants;

/**
 * @author yuqy
 *
 */
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

	@Override
	public void configureMessageBroker(MessageBrokerRegistry config) {
		config.enableSimpleBroker(RaffleConstants.RAFFLE);
		config.setApplicationDestinationPrefixes(RaffleConstants.APP_PREFIX);
	}

	@Override
	public void registerStompEndpoints(StompEndpointRegistry registry) {
		registry.addEndpoint(RaffleConstants.ENDPOINT).setAllowedOrigins("*").withSockJS();
	}

}